import torch
import numpy as np
loss = np.mean((np.array([1,2])-np.array([3,4]))**2)
print(np.array([1,2]))
print(loss)